
#include "DCTTest.h"
#include <string>

int main(int argc, char ** argv)
{
    unsigned int dim = 4096;
	if (argc >= 2)
	{
		dim = std::stoi(argv[1]);
	}
    std::vector<float> data(dim * dim);
    for (unsigned int r = 0; r < dim; ++r)
    {
        for (unsigned int c = 0; c < dim; ++c)
        {
            data[r * dim + c] = (float)rand() / (RAND_MAX - 1);
        }
    }
    std::cout << "dim is " << dim << 'x' << dim << std::endl;
    DCTTest::TestCPUDCT(data, dim);
    DCTTest::TestGPUDCT(data, dim);
    return 0;
}
